package com.kaizen.leet169;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * 169. 多数元素
 *
 * 给定一个大小为 n 的数组，找到其中的多数元素。多数元素是指在数组中出现次数大于 ⌊ n/2 ⌋ 的元素。
 *
 * 你可以假设数组是非空的，并且给定的数组总是存在多数元素。
 *
 *  
 *
 * 示例 1:
 *
 * 输入: [3,2,3]
 * 输出: 3
 * 示例 2:
 *
 * 输入: [2,2,1,1,1,2,2]
 * 输出: 2
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/majority-element
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 *
 * @author kaizen
 * @date 2020/5/9 10:13
 */
public class SolutionKaizen {

    /**
     * hash表存储计数
     * 时间复杂度O(n+n) = O(n)
     * 空间复杂度O(n)
     * 	11 ms 	45.5 MB
     * @param nums
     * @return
     */
    public int majorityElement1(int[] nums) {
        HashMap<Integer, Integer> countMap = new HashMap<>(nums.length);
        for (int num : nums) {
            countMap.compute(num, (k, v) -> {
                if (v == null) {
                    v = 1;
                } else {
                    v += 1;
                }
                return v;
            });
        }
        int freq = nums.length/2;
        for (Map.Entry<Integer, Integer> entry : countMap.entrySet()) {
            if (entry.getValue() > freq) {
                return entry.getKey();
            }
        }
        return 0;
    }

    /**
     * 	14 ms 	43.8 MB
     * @param nums
     * @return
     */
    public int majorityElement2(int[] nums) {
        int freq = nums.length/2;
        HashMap<Integer, Integer> countMap = new HashMap<>(nums.length);
        AtomicInteger res = new AtomicInteger(0);
        AtomicInteger count = new AtomicInteger();
        for (int num : nums) {
            countMap.compute(num, (k, v) -> {
                if (v == null) {
                    v = 1;
                } else {
                    v += 1;
                }
                if (count.incrementAndGet() > freq) {
                    if (v > freq) {
                        res.set(num);
                    }
                }
                return v;
            });

        }
        return res.get();
    }


}
